#!/usr/bin/python3
# -*- coding: utf-8 -*-

"""
# @version: v1.0
# @author : wlis
# @Email : 1259319710@qq.com
# @Project : g-carbon-bio
# @File : stocksz_service.py
# @Software: PyCharm
# @time: 2025/2/5 10:26
# @description : 
"""
from service.stock_base_service import StockBaseService
from service.models.stock_sz import StockSz


class StockSzService(StockBaseService):
    def __init__(self):
        self.table_name_am = 'stocksz_am'
        self.table_name_pm = 'stocksz_pm'
        super().__init__()

    def get_distinct_dates_am(self):
        return super().get_distinct_dates(self.table_name_am)


    def get_am(self, _stock_code, current_time=None, current_ap=None):
        """获取深圳当日成交量表中的股票信息

        参数:
        _stock_code: 股票代码
        current_time: 查询日期，默认为None
        current_ap: 上午或下午标识，默认为None

        返回:
        股票对象列表
        """
        return super().get(_stock_code, self.table_name_am, StockSz, current_time, current_ap)

    def get_pm(self, _stock_code, current_time=None, current_ap=None):
        """获取深圳盘后成交量表中的股票信息

        参数:
        _stock_code: 股票代码
        current_time: 查询日期，默认为None
        current_ap: 上午或下午标识，默认为None

        返回:
        股票对象列表
        """
        return super().get(_stock_code, self.table_name_pm, StockSz, current_time, current_ap)



    def add_list_am(self, stock_list):
        """将股票数据添加到深圳当日成交量表

        参数:
        stock_list: 股票对象列表
        """
        self.add_list(stock_list, self.table_name_am)

    def add_list_pm(self, stock_list):
        """将股票数据添加到深圳盘后成交量表

        参数:
        stock_list: 股票对象列表
        """
        self.add_list(stock_list, self.table_name_pm)


    def update_list_am(self, stock_list):
        """更新深圳当日成交量表中的股票数据

        参数:
        stock_list: 股票对象列表
        """
        self.update_list(stock_list, self.table_name_am)

    def update_list_pm(self, stock_list):
        """更新深圳盘后成交量表中的股票数据

        参数:
        stock_list: 股票对象列表
        """
        self.update_list(stock_list, self.table_name_pm)
